package backtracking;

import java.util.ArrayList;
import java.util.List;

/**
 * 216.组合总和|||      --->    候选者集合的元素不可以重复选取、候选者集合内不重复
 * 找出所有相加之和为 n 的 k 个数的组合，且满足下列条件：
 * ● 只使用数字1到9
 * ● 每个数字 最多使用一次
 * 返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次，组合可以以任何顺序返回。
 */
public class combinationSum3 {
    /**
     * 一刷
     */
    public static List<List<Integer>> combinationSum3(int k, int n) {
        List<List<Integer>> res = new ArrayList<List<Integer>>();
        List<Integer> list = new ArrayList<>();
        backtrack(k, n, 1, n, res, list);
        return res;
    }

    public static void backtrack(int k, int n, int start, int sum, List<List<Integer>> res, List<Integer> list) {
        if (list.size() == k && sum == 0) {
            res.add(new ArrayList<>(list));
            return;
        }

        for (int i = start; i <= 9; i++) {
            if(list.size()>k){
                break;  //剪枝
            }
            list.add(i);
            backtrack(k, n, i + 1,sum-i, res, list);
            list.remove(list.size() - 1);
        }
    }

    /**
     * 二刷跟一刷思路一样，只是实现上略有差别
     */
    class Solution {
        public List<List<Integer>> combinationSum3(int k, int n) {
            List<Integer> path=new ArrayList<>();
            List<List<Integer>> res=new ArrayList<>();
            backtracking(k,n,0,1,path,res);
            return res;
        }

        public void backtracking(int k, int n, int sum, int start, List<Integer> path, List<List<Integer>> res) {
            if (path.size() == k && sum == n) {
                res.add(new ArrayList(path));
                return;
            }

            for (int i = start; i <= 9; i++) {
                path.add(i);
                backtracking(k, n, sum + i, i + 1, path, res);
                path.remove(path.size() - 1);
            }
        }
    }
}
